package 代码随想录.哈希表;

import java.util.*;

public class _3两个数组的交集 {
    /**
     * 349. 两个数组的交集
     * https://leetcode.cn/problems/intersection-of-two-arrays/
     * 给定两个数组 nums1 和 nums2 ，返回 它们的交集 。输出结果中的每个元素一定是 唯一 的。我们可以 不考虑输出结果的顺序 。
     */
    public int[] intersection(int[] nums1, int[] nums2) {
        if (nums1 == null || nums1.length == 0 || nums2 == null || nums2.length == 0) {
            return new int[0];
        }
        Set<Integer> set1 = new HashSet<>();
        Set<Integer> resSet = new HashSet<>();
        //遍历数组1
        for (int i : nums1) {
            set1.add(i);
        }
        //遍历数组2的过程中判断哈希表中是否存在该元素
        for (int i : nums2) {
            if (set1.contains(i)) {
                resSet.add(i);
            }
        }
        //方法1：将结果集合转为数组
//        return resSet.stream().mapToInt(x -> x).toArray();

        //方法2：另外申请一个数组存放setRes中的元素,最后返回数组
        int[] arr = new int[resSet.size()];
        int j = 0;
        for(int i : resSet){
            arr[j++] = i;
        }

        return arr;
    }

    /**
     * 350. 两个数组的交集 II
     * https://leetcode.cn/problems/intersection-of-two-arrays-ii/
     * 给你两个整数数组 nums1 和 nums2 ，请你以数组形式返回两数组的交集。
     * 返回结果中每个元素出现的次数，应与元素在两个数组中都出现的次数一致（如果出现次数不一致，则考虑取较小值）。可以不考虑输出结果的顺序。
     */
    public int[] intersect(int[] nums1, int[] nums2) {
        HashMap<Integer,Integer> map = new HashMap<>();
        ArrayList<Integer> res = new ArrayList<>();
        for(int num : nums1){
            map.put(num,map.getOrDefault(num,0)+1);
        }
        for(int num : nums2){
            if(map.containsKey(num)){
                res.add(num);
                map.put(num,map.get(num)-1);
                if(map.get(num)==0)
                    map.remove(num);
            }
        }
        return res.stream().mapToInt(x -> x).toArray();
    }
}
